package qjc.b_双指针;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class c_三数之和 {
    /**
     * 初始化结果列表，排序，遍历数组，如果当前数字大于0，后续的数字都大于0，
     * 不可能再有三数之和为0，跳过重复的数字，避免重复结果
     * 定义左右指针，如果三数之和为0，加入结果列表，跳过重复的数字，避免重复结果
     * 如果三数之和小于0，移动左指针，如果三数之和大于0，移动右指针
     */
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        if(nums == null || nums.length < 3){
            return res;
        }
        for (int i = 0; i < nums.length; i++) {
            if(nums[i] > 0){
                return res;
            }
            if(i > 0 && nums[i] == nums[i-1]){
                continue;
            }
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right){
                int sum = nums[i] + nums[left] + nums[right];
                if(sum == 0){
                    res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    while (left<right && nums[left] == nums[left+1]){
                        left++;
                    }
                    while (left<right && nums[right] == nums[right-1]){
                        right--;
                    }
                    left++;
                    right--;
                }
                else if(sum < 0){
                    left++;
                }
                else if(sum > 0){
                    right--;
                }
            }
        }
        return res;
    }
}
